package org.wkg.lesson08;


import java.util.concurrent.BlockingQueue;
import java.util.concurrent.PriorityBlockingQueue;
import java.util.concurrent.TimeUnit;

/**
 * 自定义定时器
 *
 * @Author 比特就业课
 */


public class MyTimer_All {
    private BlockingQueue<MyTask_All> queue = new PriorityBlockingQueue<>();

    private Object locker = new Object();

    public MyTimer_All() {

        // 创建扫描线程
        Thread thread = new Thread(() -> {
            // 不断地扫描队列中的任务
            while (true) {
                try {
//                    System.out.println("实例化出来的对象 "+ this);
                    // 1. 从队列中取出任务
                    MyTask_All task = this.queue.take();
                    // 2. 判断到没到执行时间
                    long currentTime = System.currentTimeMillis();
                    if (currentTime >= task.getTime()) {
                        // 时间到了，执行任务
                        task.getRunnable().run();
                    } else {
                        // 当前时间与任务执行时间的差
                        long waitTime = task.getTime() - currentTime;
                        // 没有到时间，重新放回队列
                        queue.put(task);
                        synchronized (locker) {
                            // 加入等时间
                            locker.wait(waitTime);
                        }
                    }
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }

        }, "scanThread");
        // 启动线程，真正去系统中申请资源
        thread.start();

        Thread daemonThread = new Thread(() -> {
            while (true) {
                synchronized (locker) {
                    locker.notifyAll();
                }
                try {
                    TimeUnit.MILLISECONDS.sleep(10);
                } catch (InterruptedException e) {
                    throw new RuntimeException(e);
                }
            }
        });
        daemonThread.setDaemon(true);
        daemonThread.start();


    }

    /**
     * 添加任务的方法
     *
     * @param runnable 任务
     * @param delay    延时
     * @throws InterruptedException
     */
    public void schedule(Runnable runnable, long delay) throws InterruptedException {
        // 根据传处的参数，构造一个MyTask
        MyTask_All task = new MyTask_All(runnable, delay);
        // 把任务放入阻塞队列
        queue.put(task);
        //System.out.println("添加成功");
        // 唤醒等待的线程
        synchronized (locker) {
            locker.notifyAll();
        }
    }


}

// 用一个类来描述任务及任务执行的时间
class MyTask_All implements Comparable<MyTask_All> {
    // 任务
    private Runnable runnable;
    // 任务执行的时间
    private long time;

    public MyTask_All(Runnable runnable, long delay) {
        // 校验任务不能为空
        if (runnable == null) {
            throw new IllegalArgumentException("任务不能为空.");
        }
        // 时间不能为负数
        if (delay < 0) {
            throw new IllegalArgumentException("执行时间不能小于0.");
        }
        this.runnable = runnable;
        // 计算出任务执行的具体时间
        this.time = delay + System.currentTimeMillis();
    }

    public Runnable getRunnable() {
        return runnable;
    }

    public long getTime() {
        return time;
    }

    @Override
    public int compareTo(MyTask_All o) {
        if (this.getTime() > o.getTime()) {
            return 1;
        } else if (this.getTime() < o.getTime()) {
            return -1;
        } else {
            return 0;
        }

//        return (int) (this.time - o.getTime());
    }
}

